import os

import flask
from flask import request

app = flask.Flask(__name__)


@app.route("/upload_images", methods=["GET", "POST"])
def upload_images():
    if request.method == "POST":
        file = request.files["file"]
        # hostname = request.args.get("hostname")
        dir_name = f"./static/boss_images"
        if not os.path.exists(dir_name):
            os.mkdir(dir_name)
        file_name = f"{dir_name}/" + file.filename
        print("收到文件：", file_name)
        file.save(file_name)
        return "success"
    else:
        return "success get"


@app.route("/query_images", methods=["GET"])
def query_images():
    dir_name = "./static/boss_images"
    html = "<html><body><ul>"
    fnames = sorted([fname for fname in os.listdir(dir_name)
              if fname.endswith(".png")], reverse=True)
    for fname in fnames[:10]:
        html += f"""
            {fname}: <br/>
            <img width='80%' src='{dir_name}/{fname}' />
            <br /><br />
        """
    html += "</ul></body></html>"
    return html


app.run("0.0.0.0", 8888)
